from setuptools import setup, find_packages
import os, shutil

CWD = os.path.dirname(os.path.abspath(__file__))
path = os.path.join(CWD, "build")
if os.path.isdir(path):
    print("INFO del dir ", path)
    shutil.rmtree(path)

setup(
    name="Torlink",
    description="Streaming ML Using PyTorch",
    author="TCSE@ISCAS",
    packages=find_packages(),
    version="1.0.0",
    install_requires=[
        "torch>=2.1.0",
        "torchvision>=0.16.0",
        "numpy>=1.23.1",
        "scikit-learn>=1.3.2",
        "typing_extensions>=4.8.0",
        "redis>=5.0.1",
    ],
)
